# @Time : 2024/6/13 14:50
# @Author : ZHUYI
# @File : DB_connection
import pymysql


# 建立数据库连接并创建表格
def db_connection():
    """
    连接到MySQL数据库，并创建一个名为juejin的表。

    返回:
    - conn: 数据库连接对象
    - cursor: 数据库游标对象
    """
    # 连接数据库，设置数据库地址、端口、用户名、密码和数据库名称
    conn = pymysql.connect(host="127.0.0.1", port=3306, user='root', passwd='123456', db='testpython',
                           charset='utf8mb4')
    cursor = conn.cursor()

    # 如果juejin表已存在，则删除该表
    cursor.execute("DROP TABLE IF EXISTS juejin")

    # 创建juejin表的SQL语句，定义表的结构和字段注释
    sqlCreat = """CREATE TABLE juejin (
                     id VARCHAR(255) NOT NULL primary key COMMENT '掘金小册id',
                     title  VARCHAR(255) NOT NULL COMMENT '小册标题',
                     summary  VARCHAR(255) NOT NULL COMMENT '小册介绍',
                     author  VARCHAR(255) NOT NULL COMMENT '小册作者',
                     job_title  VARCHAR(255) NOT NULL COMMENT '作者头衔',
                     cover_img  VARCHAR(255) NOT NULL COMMENT '小册图片链接',
                     price INT NOT NULL COMMENT '小册价格',
                     buy_count INT NOT NULL COMMENT '销售数量',
                     is_finished INT NOT NULL COMMENT '是否完结(0:正在更新，1:已完结)'
                     )"""

    # 执行创建表的SQL语句
    cursor.execute(sqlCreat)

    # 返回数据库连接和游标对象
    return conn, cursor



